vendor: str = "nvidia"
train_batch_size = 2
eval_batch_size = 2
